import pyvista as pv
import time

pv.start_xvfb()

plotter = pv.Plotter(off_screen=True)
mesh = pv.read('outputs_control_arm/inverse/2025-05-24/06-33-57/visual/epoch_100/vis.vtu')

plotter.add_mesh(mesh, show_edges=True)
plotter.view_xy()

plotter.show(screenshot="output.png")